"""
汽车交通模拟的主程序。

嗯
"""
import numpy as np
import sub_function as sub
# =============================================================================
# 供修改的参数列表
END_TIME = 10  # 分钟(min)
TIME_INTERVAL = 0.1  # 时间间隔取0.1秒，运动员的反应速度
CAR_NUM=3

# =============================================================================
TIME_SHEET = np.arange(0, END_TIME * 60, dtype=np.int64)  # 时间序列

car_data = np.zeros((CAR_NUM, 2, np.size(TIME_SHEET)))


# 初始条件
for i in np.arange(0, CAR_NUM):  # 赋初值，按照不同行的车，赋予列上的位置和速度
    car_data[i][0][0] = i  # 赋予初位移position
    car_data[i][1][:] = i+1  # 赋予初速度speed 

# 时间迭代
car_data = sub.move(TIME_SHEET, TIME_INTERVAL, car_data)
# 制作数组切片，增加可读性。
car_data_pos = car_data[:, 0, :]
car_data_speed = car_data[:, 1, :]

sub.driving_plot(car_data_pos)



